import warnings

import matplotlib.pyplot as plt
import pandas as pd

warnings.filterwarnings('ignore')
# 1.将其中“Stu_Name”字段设计成索引,使用工具读取"stu.txt"文件。
df = pd.read_csv('stu.txt',sep=' ')
stu_name = df['Stu_Name']
# print(df)
df2 = df.drop(['Stu_Name'],axis=1)
df2.set_index(stu_name,inplace=True)
print(df2)
# 2.设计把每个人的工资上调20% 展示,并新增一列数据"Yonns"。
df2['Yonns'] = df2['money'] * 1.2
# 3.用pandas算子按年龄段统计不同城市的money总金额，自定义设置字段类型。
bins = [0,30,40]
df2['年龄段'] = pd.cut(df2['Stu_Age'], bins=bins,labels=['0-30','30-40'])
result3 = df2.groupby(['City','年龄段'])['money'].sum().reset_index()
print(result3)
# 4.使用pandas可视化API绘图，要求按照城市统计不同年龄 总金额。
result4 = df2.groupby(['City','年龄段'])['money'].sum().reset_index()
print(result4)
for i in result4['City'].unique():
    cc = result4[result4['City'] == i]
    plt.plot(cc['年龄段'],cc['money'])
plt.show()